"""
Author : sunshicheng
DateTime : 2021/3/10.2:01 下午
FileName : locust_test.py
Desc: 尝试使用locust 进行简单的性能测试

继承任务类Task
和虚拟用户类 HttpUser
"""

import sys
from locust import TaskSet, HttpUser, task, between, constant

sys.path.append('..')
from config.base_info import Static_info as si
from config.user_token import Token

token = Token('18226287291')
headers = token.test_token()


class Mytask(TaskSet):
    @task
    def get_test(self):
        data = {"account": "sunshicheng"}

        # 发送get请求，加上name，就会展示传入的name名称
        result = self.client.get(si.inter['saas']['saas_permission'], params=data, headers=headers, name='获取用户权限')
        print(result.text)


# 创建虚拟用户类
class webUsers(HttpUser):
    weight = 3
    # 需要执行的任务
    tasks = [Mytask]
    # wait_time 可以有三种写法，一个是随机时间between，一个是固定时间constant,还可以自己定义一个时间函数
    wait_time = constant(1)

# class mobileUsers(HttpUser):
#     weight = 6
#     # 需要执行的任务
#     tasks = [Mytask]
#     # wait_time 可以有三种写法，一个是随机时间between，一个是固定时间constant,还可以自己定义一个时间函数
#     wait_time = constant(1)
